/*
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

/*
 *    RemoteExperimentSubTask.java
 *    Copyright (C) 2000-2012 University of Waikato, Hamilton, New Zealand
 *
 */

package weka.experiment;

import java.io.File;

import weka.core.SerializedObject;

/**
 * Class to encapsulate an experiment as a task that can be executed on a remote
 * host.
 * 
 * @author Mark Hall (mhall@cs.waikato.ac.nz)
 * @version $Revision$
 */
public class RemoteExperimentSubTask implements Task {

    /** ID added to avoid warning */
    private static final long serialVersionUID = -1674092706571603720L;

    /* Info on the task */
    private TaskStatusInfo m_result = new TaskStatusInfo();

    /* The (sub) experiment to execute */
    private Experiment m_experiment;

    private SerializedObject m_serializedExp;

    public RemoteExperimentSubTask() {
        m_result.setStatusMessage("Not running.");
        m_result.setExecutionStatus(TaskStatusInfo.TO_BE_RUN);
    }

    /**
     * Set the experiment for this sub task
     * 
     * @param task the experiment
     */
    public void setExperiment(Experiment task) {
        // m_experiment = task;
        try {
            m_serializedExp = new SerializedObject(task);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * Get the experiment for this sub task
     * 
     * @return this sub task's experiment
     */
    public Experiment getExperiment() {
        if (m_experiment != null) {
            return m_experiment;
        } else {
            m_experiment = (Experiment) m_serializedExp.getObject();
            return m_experiment;
        }
    }

    /**
     * Run the experiment
     */
    @Override
    public void execute() {
        m_experiment = (Experiment) m_serializedExp.getObject();
        // FastVector result = new FastVector();
        m_result = new TaskStatusInfo();
        m_result.setStatusMessage("Running...");
        String goodResult = "(sub)experiment completed successfully";
        String subTaskType;
        if (m_experiment.getRunLower() != m_experiment.getRunUpper()) {
            subTaskType = "(dataset " + ((File) m_experiment.getDatasets().elementAt(0)).getName();
        } else {
            subTaskType = "(exp run # " + m_experiment.getRunLower();
        }
        try {
            System.err.println("Initializing " + subTaskType + ")...");
            m_experiment.initialize();
            System.err.println("Iterating " + subTaskType + ")...");
            // Do not invoke runExperiment(): every exception will be lost
            while (m_experiment.hasMoreIterations()) {
                m_experiment.nextIteration();
            }
            System.err.println("Postprocessing " + subTaskType + ")...");
            m_experiment.postProcess();
        } catch (Exception ex) {
            ex.printStackTrace();
            String badResult = "(sub)experiment " + subTaskType + ") failed : " + ex.toString();
            m_result.setExecutionStatus(TaskStatusInfo.FAILED);
            // m_result.addElement(new Integer(RemoteExperiment.FAILED));
            // m_result.addElement(badResult);
            m_result.setStatusMessage(badResult);
            m_result.setTaskResult("Failed");
            // return m_result;
            return;
        }
        // m_result.addElement(new Integer(RemoteExperiment.FINISHED));
        // m_result.addElement(goodResult);
        m_result.setExecutionStatus(TaskStatusInfo.FINISHED);
        m_result.setStatusMessage(goodResult + " " + subTaskType + ").");
        m_result.setTaskResult("No errors");
        // return m_result;
    }

    @Override
    public TaskStatusInfo getTaskStatus() {
        return m_result;
    }

}
